package com.future;

import com.future.util.TreeNode;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * Description: 剑指 Offer 32 - III. 从上到下打印二叉树 III
 * 请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。
 * <p>
 * 例如:
 * 给定二叉树: [3,9,20,null,null,15,7],
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * Author: weiruibai.vendor
 * Date: 2022/7/23 23:18
 */
public class Solution_32_III {

    public static void main(String[] args) {

        TreeNode head = new TreeNode(3);
        head.left = new TreeNode(9);
        head.right = new TreeNode(20);
        head.right.left = new TreeNode(15);
        head.right.right = new TreeNode(7);

        System.out.println(levelOrder(head));
    }


    /**
     * 主流程就是从上到下，从左到右，当标识表示从右往左是就把该层反转
     *
     * @param root
     * @return
     */
    public static List<List<Integer>> levelOrder(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        List<List<Integer>> ans = new ArrayList<>();
        boolean fromLeft = true;
        LinkedList<TreeNode> linkedList = new LinkedList<>();
        linkedList.add(root);
        TreeNode nextLine = root.left != null ? root.left : root.right;
        List<Integer> tmpAns = new ArrayList<>();
        while (!linkedList.isEmpty()) {
            TreeNode node = linkedList.pop();
            if (node == nextLine) {
                if (!fromLeft) {
                    /**
                     * 当从右往左时，需要把顺序反转
                     */
                    List<Integer> tmp = new ArrayList<>();
                    for (int i = tmpAns.size() - 1; i >= 0; i--) {
                        tmp.add(tmpAns.get(i));
                    }
                    tmpAns = tmp;
                }
                ans.add(tmpAns);
                tmpAns = new ArrayList<>();
                fromLeft = !fromLeft;
                /**
                 *          4
                 *        /  \
                 *       1    8
                 *           / \
                 *          10  90
                 * 当遍历到值1时，1是叶子节点，导致nextLine=null
                 * 所以需要(2)(3)弥补把nextLine=10
                 *
                 * （1）
                 * 注意：
                 * 这里可能也为空，因为node可能就是叶子节点，针对这种场景，需要在（2）,（3）补充
                 */
                nextLine = node.left != null ? node.left : node.right;
            }
            tmpAns.add(node.val);
            if (node.left != null) {
                linkedList.add(node.left);
                /**
                 * （2）
                 * 解决（1）中为null的情况
                 */
                nextLine = nextLine != null ? nextLine : node.left;
            }
            if (node.right != null) {
                linkedList.add(node.right);
                /**
                 * （3）
                 * 解决（1）中为null的情况
                 */
                nextLine = nextLine != null ? nextLine : node.right;
            }
        }
        if (!fromLeft) {
            // 反转tmpAns
            List<Integer> tmp = new ArrayList<>();
            for (int i = tmpAns.size() - 1; i >= 0; i--) {
                tmp.add(tmpAns.get(i));
            }
            tmpAns = tmp;
        }
        ans.add(tmpAns);
        return ans;
    }

}
